package com.ruoyi.framework.security.handle;

import java.io.IOException;
import java.io.Serializable;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.security.core.AuthenticationException;
import org.springframework.security.web.AuthenticationEntryPoint;
import org.springframework.stereotype.Component;
import com.alibaba.fastjson2.JSON;
import com.ruoyi.common.constant.HttpStatus;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.utils.ServletUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.framework.security.filter.WxAuthenticationException;

/**
 * 认证失败处理类 返回未授权
 *
 * @author ruoyi
 */
@Component
public class AuthenticationEntryPointImpl implements AuthenticationEntryPoint, Serializable {
    private static final long serialVersionUID = -8970718410437077606L;

    @Override
    public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException e)
            throws IOException {
        int code = HttpStatus.UNAUTHORIZED;
        String msg;

        // 1. 区分微信认证异常和系统认证异常
        if (e instanceof WxAuthenticationException) {
            // 微信认证异常：使用自定义消息（如"微信登录凭证已过期"）
            msg = e.getMessage();
        } else {
            // 系统原有认证异常：使用系统默认消息
            msg = "登录状态已过期，请重新登录";
        }

        // 2. 保留原逻辑：添加请求路径信息（可选，根据需要决定是否保留）
        // 注意：如果保留，会覆盖上面的msg，需要调整拼接方式
        if (StringUtils.isNotEmpty(request.getRequestURI())) {
            msg = StringUtils.format("请求访问：{}，{}", request.getRequestURI(), msg);
        }

        // 3. 统一返回JSON格式
        ServletUtils.renderString(response, JSON.toJSONString(AjaxResult.error(code, msg)));
    }
}